diff --git a/src/remote_generator/services/entity_generator.rb b/src/remote_generator/services/entity_generator.rb
index 454be98..7ac754e 100644
--- a/src/remote_generator/services/entity_generator.rb
+++ b/src/remote_generator/services/entity_generator.rb
@@ -141,8 +141,7 @@ def attributes_types_union
         end
 
         def association_property_names_ts_array
-          binding.pry
-          attribute_names.inspect
+          associations.keys.map(&:to_s).inspect
         end
 
         def attribute_names