Fix Multistage argument parsing

Fix #297