error fix

Issue #1 error fix