Fix build with missing include