Fix for disappearing units when following another unit