Fixed up scaling issue since input data is different