Fix missed tests