Fix import of setup_logging