Alter example according to github comments