Fixing a little oversight