Fix AI movement