Fix broken assertion for cli module