adding missing include

Without the include I had a compilation error