Moved test to after argument parsing