Fix a typo in testing refactor. Related to #333