Fixed a regression bug