Little fix for test.