Add forgotten import