Fixing Python class snippet.