Fix for dataset graphing