Yeah I can move it to the class. I'll do the same for `MigrateTests`.