[Solved][1.16.1] Flying Entity with Follow Goal

[DavidQF555]

I am creating a flying Custom Entity whom will follow an owner, but it doesn't seem to want to move up ever. This is the entity class and goal class: 

public class CustomEntity extends FlyingEntity {

	private LivingEntity owner;

	public CustomEntity(EntityType<? extends FlyingEntity> type, World worldIn, LivingEntity owner) {
		super(type, worldIn);
		this.owner = owner;
		this.navigator = new FlyingPathNavigator(this, worldIn);
	}
  
  @Override
	public void registerGoals() {
		super.registerGoals();
		goalSelector.addGoal(0, new FollowOwnerGoal());
	}

private class FollowOwnerGoal extends Goal {

		private static final int DISTANCE = 5;

		public FollowOwnerGoal() {
			this.setMutexFlags(EnumSet.of(Goal.Flag.MOVE));
		}

		@Override
		public boolean shouldExecute() {
			return CustomEntity.this.owner != null && CustomEntity.this.getDistanceSq(CustomEntity.this.owner) > DISTANCE * DISTANCE;
		}
		
		@Override
		public boolean shouldContinueExecuting() {
			return !CustomEntity.this.getNavigator().noPath() && CustomEntity.this.getDistanceSq(CustomEntity.this.owner) > DISTANCE * DISTANCE;
		}
		
		@Override
		public void resetTask() {
			CustomEntity.this.getNavigator().clearPath();
		}

		@Override
		public void tick() {
			super.tick();
			LivingEntity owner = CustomEntity.this.owner;
			CustomEntity.this.getNavigator().tryMoveToEntityLiving(owner, 5);
		}
	}
}

I thought that just making the navigator a FlyingPathNavigator would work, but it doesn't move upward for navigation. Is there a way to solve this or do I need to write a navigator myself?

Edited August 1, 2020 by DavidQF555

[DavidQF555]

Solved, you have to change the moveController to FlyingMovementController