Added the extension to the test script