Fix comment
Clean up