minor change to gpustats function