Can use ```assertEqual``` here