Added missing include file, fixes #6