Fixing smoke test