Adds last missing test cases for code coverage