Issue #11 fixed

Am reparat eroarea.