fix a potential bug