Fix missing include
Merge pull request #1 from Pronovix/fix/missing-include

Fix missing include