Fixed failing tests for extension removal