Small fix
Merge pull request #2 from zenedge/some_improv

Small fix