Bug fix: fix invalid argument