Make sure TM automation still works

(Redstone) automation used to work in 1.20.1. This needs double checking post port to 1.21.1 and potentially fix any issues.