Fix required argument bug