Fix missing object reference