Add assertion for CLI output