bug fix
Closed #253