Fix missed case