Test mode fix - allow graphing of model