fix a possible bug