Fixing compile

Closes issue #2