Fix the testing