Update so it works for 1.1 and up
Merge pull request #2 from nibrobb/patch-2

Update so it works for 1.1 and up