Fix compile bug we introduced!