Fixed method call in tests