Fix example test code