Fixed argument mismatch bugs