Migrating tests out of extension