Tridents don't return from the void

I know for a fact that in a past update (1.15.x or something) Mojang made it so that tridents with loyalty 1, 2 or 3 return when thrown in the void. In this snapshot that wasn't the case for my beloved god-tier trident.