diff --git a/lib/custom_wizard/builder.rb b/lib/custom_wizard/builder.rb
index 6084ad5f..a44b1192 100644
--- a/lib/custom_wizard/builder.rb
+++ b/lib/custom_wizard/builder.rb
@@ -256,8 +256,8 @@ class CustomWizard::Builder
           with_type: true
         }
       ).perform
-                  
-      if content[:type] == 'association'
+                        
+      if content && content[:type] == 'association'
         content[:result] = content[:result].map do |item|
           { 
             id: item[:key],