Fixing a pretty big oversight from the previous commit.