Add forgotten schema updates