Fix for uncompilable master

Fixes #1