Quick fix on linear fit