Missing include

Added missing include, fixing warning when compiling on gcc.