Fix missed argument