Little fix after merging

So close...