Fix missing include