bot: Fix for new model