Fixes build error

Closes #6