Fixed arg bug