Fix the first test code