Add small fix for an example