Fix build errors

Fixes #2.