forgot header files (see previous commit)