Remove test code that depends on extension