Annotate tests
check level
fix extend
Merge pull request #94 from TheSwanFactory/fix-tests

Fix tests