Added missing module importation