Update tests to pass lint checks