Fixed error in test