fix argument improvement

Fix several errors caused by #114

Signed-off-by: Hiroshi Miura <miurahr@linux.com>