Fixed bug after migrating MCMC to Lora