diff --git a/lib/generators/jsonapi/swagger/swagger_generator.rb b/lib/generators/jsonapi/swagger/swagger_generator.rb
index 65408c4..f8b53a5 100644
--- a/lib/generators/jsonapi/swagger/swagger_generator.rb
+++ b/lib/generators/jsonapi/swagger/swagger_generator.rb
@@ -82,7 +82,7 @@ module Jsonapi
     end
 
     def model_klass
-      model_class_name.safe_constantize
+      file_name.camelize.safe_constantize
     end
 
     def resource_klass