Fixed method argument bug