Small code fix
Merge pull request #90 from hwdro/master

Small code fix.