Fix small argument bug