my $plan;

sub VERSION {
    $plan = $_[1];
}

sub import {
    my $self = shift(@_);
    goto &UpdateTests   if  ! @_;
    my $module = shift(@_);
    my @plan = "no_plan";
    if(  defined($plan)  ) {
        @plan = ( tests => $plan );
    } elsif(  "Test::Simple" eq $module  ) {
        @plan = ( tests => 0 );
    }
    unshift @_, $module, @plan;
    my $import = $module . "::import";
    undef $plan;
    goto &$import;
}

sub UpdateTests {
    for my $test ( @ARGV ) {
        my $plan = RunTestsAndCount( $test );
        @ARGV = $test;
        $^I = ".old";
        while( <> ) {
            s{
                ^(
                  [\w\s]*
                  (?<![\w:])
                  Test::AutoPlan
                )
                (\s+\d+)?
            }{
                $1 . " " . $plan
            }ex;
            print;
        }
    }
}