better fix for aphyr/riemann#233