Complete this after we merge: https://github.com/SREGym/SREGym/pull/395
Complete this after we merge: SREGym/SREGym#395