Fix the bug that the token after atom is ignored