Small fix
Merge pull request #8 from thomassjogren/dev

Small fix