Update missed imports