Fix another linting issue and add tests for a method that I added and forgot about