fixed  bug (line 80)