Fix Lora transition issues