Fixed handling arguments