Fix pyling warning