Minor bug fix

Solved the problem causing unexpected end of file error.
Merge pull request #11 from saji89/patch-1

Minor bug fix