カテゴリ毎に特定のコードを記述したい(アフィンガー6)

/* カテゴリ別アフィリエイト */
function custom_affiliate_widget() {
    ob_start(); // 出力をバッファリング

    if (in_category('HTML')) {
        echo '<div class="afi-content">
			　<!-- HTMLカテゴリー用のアフィリエイトコードを記述 -->
              </div>';
    } elseif (in_category('windows')) {
        echo '<div class="afi-content">
			　<!-- HTMLカテゴリー用のアフィリエイトコードを記述 -->
              </div>';
    } else {
        echo '<div class="rakuten-afi">	
			 <!-- HTMLカテゴリー用のアフィリエイトコードを記述(その他のカテゴリー) -->
			 </div>';
	}

    return ob_get_clean(); // バッファの内容を返す
}
add_shortcode('affiliate_widget', 'custom_affiliate_widget');

[affiliate_widget]