Small but rather important fix
Merge pull request #2 from robert-stuttaford/patch-1

Small but rather important fix