Change migration to work with postgres