fix mistake from last commit