Improve AI Catalog feature tests