Fix type_traits check