Fix build break
Merge pull request #185 from edrevo/fix-build-break

Fix build break