fix type signature of axle.visualize.play