fix a BUG