Fixed errors on unit tests