Add test for extension