oops, broke gym with earlier changes