Fix up that code snippet