Fix broken smoke test