商品詳細ページのバリエーション画像について

現在、『Be Yours』というShopifyテーマでセットアップしていますが、バリエーションごとに複数の商品画像を持つ機能を実装しました。
（商品バリエーションのカラーを選択すると、該当カラーの商品画像のみが表示される）
以下の通りです。

しかし、読み込んだ時点では全ての画像が表示されているので、読み込んだ時点で該当カラーのものだけが表示されるようにしたい。

分かる方がいれば教えてほしいです(>_<)