Fixed error

I believe this should fix the error showing up on the page
Merge pull request #7 from rgoomar/patch-2

Fixed error