Fixed code snippet