Fix build

Fixes #1