Fix some missed updates