Fixing issue in regression