Fixing another argument issue