fixed bug that was preventing treatment variable from being output