Fixed argument bug