fixing tests to register by type