Fixing migration issue where values are none