Fix path
Add more tests to increase coverage to 90%