Fixed wrong argument bug