adds minor code correction for count