fix done.
Merge pull request #1 from msusur/master

Code fix