Merge pull request #46 from dizzzz/develop

Minor code change