Add missing include, and make GCC happy