Fix bug where it will run more tests than desired
Release 0.0.2