Fixed errors on tests