Issue #166 - fix code