Fix sidestep in mlx.