Fix bug where test would pass but raise AssertionError

Simple bug fix in `wip` decorator