Small fix
Merge pull request #3 from hgslabs/v2.0

Small fix