fixed a regression bug