Fix missed var