Fix restrictions with branches in broadcast variables and iteration closures