Fixed the testing problem