lib/adeia/controller_resource.rb in adeia-0.11.6 vs lib/adeia/controller_resource.rb in adeia-0.11.7

@@ -80,13 +80,17 @@
     end
 
     private
 
     def resource_class
-      begin
-        @controller_name.classify.constantize
-      rescue NameError
-        @controller_name.classify.demodulize.constantize
+      if @resource
+        @resource.class
+      else
+        begin
+          @controller_name.classify.constantize
+        rescue NameError
+          @controller_name.classify.demodulize.constantize
+        end
       end
     end
 
     def resource_name
       @model.model_name.element