Should fix the test coverage
Merge pull request #329 from ZeusWPI/coverage/fix-coverage

Should fix the test coverage