slight fix, test count