Fix wrong code in unittest example.