Add some missed changes