fix mistake when refactoring #1a3b8afb73a3fc9d4db5d55a7d0182bb8119a2cc