Fixed error in test code