minor fix to run on data