Fixing Code Snippet