fix small test bug