diff --git a/extensions/ipec/models/ipec_exploits.rb b/extensions/ipec/models/ipec_exploits.rb
index 4c48589e3..1c4dcc46b 100644
--- a/extensions/ipec/models/ipec_exploits.rb
+++ b/extensions/ipec/models/ipec_exploits.rb
@@ -6,13 +6,9 @@
 module BeEF
   module Core
     module Models
-      class IpecExploits < ActiveRecord::Base
-        attribute :id, :Serial
-        attribute :name, :Text, :lazy => false
-        attribute :protocol, :String, :lazy => false
-        attribute :os, :String, :lazy => false
+      class IpecExploits < BeEF::Core::Model
 
-        belongs_to :extension_ipec_exploits_run, 'IpecExploitsRun'
+        has_many :ipec_exploits_run
 
       end
 
diff --git a/extensions/ipec/models/ipec_exploits_run.rb b/extensions/ipec/models/ipec_exploits_run.rb
index c87f7aed8..bc7616616 100644
--- a/extensions/ipec/models/ipec_exploits_run.rb
+++ b/extensions/ipec/models/ipec_exploits_run.rb
@@ -6,14 +6,9 @@
 module BeEF
   module Core
     module Models
-      class IpecExploitsRun < ActiveRecord::Base
+      class IpecExploitsRun < BeEF::Core::Model
 
-        attribute :id, :Serial
-        attribute :launched, :Boolean, :lazy => false
-        attribute :http_headers, :Text, :lazy => false
-        attribute :junk_size, :String, :length => 3, :lazy => false
-
-        belongs_to :ipec_exploits
+        belongs_to :ipec_exploit
 
       end