Fixing context import
fixing logging import