Get proper record class for vm_infra controller

Author: Hilda Stastna

app/controllers/vm_infra_controller.rb

@@ -14,6 +14,10 @@ def self.table_name
     @table_name ||= "vm_infra"
   end
 
+  def record_class
+    VmOrTemplate
+  end
+
   private
 
   def features