fix trial constants issue when saving multi trial env
Reviewer: dan jao