Fix build
Merge pull request #367 from ovatsus/master

Fix build