[PATCH v2 1/6] test: fix test_extension.py