ENH: updated tests for coverage