Added a missing include. Thanks @linkmauve1