Fix smoke test