Fix use correct method call in test