Fix import in setup_handlers