thymeleafでのデータ表示について

実現したいこと

ここに実現したいことを箇条書きで書いてください。
thymeleafのテンプレート切り替え機能を動作させて2つのテーブルをツリーの選択によって表示を切り替えたい

Aのデータ、Bのデータをツリーノードが選択された際にテーブルのデータとして一覧表示させたかったのですがうまくいかず困っています。
th:if="{code} == 0"のようにしてAのデータがformのcodeという変数が0の時に表示されるように、
th:if="{code} != 0"のようにしてBのデータがformのcodeという変数が0以外の時に表示されるように、設定したかったのですが
ajaxでこのコードのデータをセットした際に表示が切り替わらず困っています。
リロードした際にはデータ表示されるのですがこれを解消したいです。