fix the test errors