Fix for issue #1

Add missing include to fix compile errors