Fixed errors in test