Fixing a typo and adding a missing header - thanks to fullTalgoRythm