Fix dependency

Closes #4