Fixed tests after method rename