Adds coverage tests to fix