fix method used in test