Fix broken build

Fixes issue #2