fix a potential bug (?)