fixed build break
Merge pull request #48 from OneGet/wip

fixed build break