Fixing migration dependency