fix a bug that cause test error