Fixed the Linear Regression model