Fix for missing type include in public header