Fix argument support

Signed-off-by: Levente Kurusa <levex@linux.com>