Fix merge trains checkbox

Another small fix for failure related to pajamas migration. Verified, test passed against Staging with fix.

Related to #360247 (closed)