Add new hook for two-step media type checking