Fix failing smoke test