Invalid argument bug fixed