Fixed tests for extension