Added missing include and fix for compiler warning.