Fix test count