Fixing argument mismatch