fix a bug in test again