fix little issue #211 bug