Adding migration to restore primary key - currently only set to run in development