fix #232 bug