Fix for buggy example code test