diff --git a/db/migrate/20200913092024_create_grid_weapons.rb b/db/migrate/20200913092024_create_grid_weapons.rb
index 7d4a572..34eada4 100644
--- a/db/migrate/20200913092024_create_grid_weapons.rb
+++ b/db/migrate/20200913092024_create_grid_weapons.rb
@@ -1,13 +1,14 @@
 class CreateGridWeapons < ActiveRecord::Migration[6.0]
     def change
-            t.belongs_to :composition, type: :uuid
-            
         create_table :grid_weapons, id: :uuid, default: -> { "gen_random_uuid()" } do |t|
+            t.references :party, type: :uuid
             t.references :weapon, type: :uuid
+            
             t.references :weapon_key1, class_name: 'WeaponKey', type: :uuid
             t.references :weapon_key2, class_name: 'WeaponKey', type: :uuid
 
             t.integer :uncap_level
+            t.boolean :mainhand
             t.integer :position
 
             t.timestamps