Added missing migration for tests