Fixed small bug that occured when running PHPUnit