fixed tiny bug