fix build (squashed)
Merge pull request #220 from pfalabella/fix-build

fix build