fix a regression bug