Add in missing header (thanks to  AlisterH)