fix a bug due to last update