Fix TU after pylint