fix the tiny bug.