[TMorris] Fixed the testing issue