Fix bug where test would pass but raise AssertionError
Merge pull request #103 from mandli/fix-wip-decorator

Fix bug where test would pass but raise AssertionError