Fix little oversight