fix a bug in test