Fixes test using old method.