Fixing migration script for adding fields