fixing error in test