Fixing migration for postgres