Fixing arg bug...