Custom post type fix