Fix stupid mistake from last commit