Fix coverage. Closes #1