Fix dist build

Fixes #9