Fixed code sample