fix #104

/cc @calvinfo