Вордпресс; Отображение другого шаблона контента в зависимости от категории

Я хочу взять базовый цикл WordPress и в том месте, где он вызывает шаблон контента, я хочу запустить оператор if else, который извлекает другую часть шаблона в зависимости от категории. Я знаю, что могу сделать кучу копий archive.php, но зачем делать это для каждой отдельной категории, которую вы хотите использовать?

Итак, я попробовал этот код, но он просто вытаскивал шаблон архива в каждом случае:

while ( have_posts() ) :
            the_post();

        if ( is_category( 'my-category-slug' )) {
            get_template_part( 'template-parts/content', 'single' );
        } else {
            get_template_part( 'template-parts/content', 'archive' );
        }

endwhile;

Итак, могу ли я даже запустить такой оператор if else в цикле?