quick bug fix
Merge pull request #64 from brightredchilli/quick-fix

quick bug fix