change the getCount method