Fixing migration for Travis