Small change for mongo-friendly predictions