Fixed missing import in config