Fix unit test - automigrate is async