Compilation fix. Thanks to matt7code

Fixes #11