Fix Multistage argument parsing

Close #297