Fixing function args