Fixed arg parsing bug