corrected error in test code