Fix tests not running