I tried manually overriding to get this working in 2.1 experimental and there is an error when upgrading underground belts related to this code in control.lua
if entity.type == "underground-belt" and entity.neighbours then
Seems this was renamed in 2.1, it's now entity.underground_belt_neighbour.