Fix missed diff
Add helper method