assertEquals could be used here