fixes this test helper
Improving test coverage