Fix sily mistake with last commit