Fixed a BUG in this code. 

A minor typo that could have had a BIG impact into the model performance because starting the optimization (potentially) from a bad space!