fix the bug intruduced by last commit