Fix puzzle init regression