Add condition for import