Add additional tests for mistral