Fixed errors in tests