Fixing mistake from last commit