Improve extension test
Merge pull request #32 from thewilkybarkid/extension-test

Fix extension test