Fix import in scripts.