fixing mistake from last commit