update to work with latest llama version