Fixed errors in new tests