Fixing arg errors