Minor cleanup for pylama.