bug fix
Issue #5