Update tests to pass linting checks