fix migration for testing purposes