Address PyFlakes feedback