Fixed a missing include