Fix important file
Merge pull request #4 from rajiteh/bug/serious-bug

Fix important file