WordPressにカスタム投稿タイプを増やしたときにカテゴリページで投稿一覧がでてこない

初回作る時には覚えてたり調べたりするのですが・・・しばらくたったあと、カスタム投稿タイプを追加した後、、なぜかカテゴリ一覧がでてこない。。

つい忘れてしまって、、1時間ぐらい無駄にしてしまったので、、備忘。

“pre_get_postsに、add_actionで、追加したカスタム投稿タイプを追記すること。”

function add_post_category_archive( $wp_query ) {
    if ($wp_query->is_main_query() && $wp_query->is_category()) {
        $wp_query->set( 'post_type', array('post','custom1'));
    }
}
add_action( 'pre_get_posts', 'add_post_category_archive');

投稿タイプ”custom2″を追加したとすると・・・

function add_post_category_archive( $wp_query ) {
    if ($wp_query->is_main_query() && $wp_query->is_category()) {
        $wp_query->set( 'post_type', array('post','custom1', 'custom2'));
    }
}
add_action( 'pre_get_posts', 'add_post_category_archive');

こんな感じ。