Fixed broken code snippet