revert extension to fix test