Fixing the example and a possible bug