Fix broken AI movement on Z axis