Fix stuck on treeline bug