Forgot some code in example