Fixed the tests not running