Fix coverage. Closes #195