Fixing the ROI class