Fixed error after merging in test classes.