1 files changed, 8 insertions, 2 deletions
diff --git a/lib/Reply/Plugin/Defaults.pm b/lib/Reply/Plugin/Defaults.pm
index 40dee9b..ca3c047 100644
--- a/lib/Reply/Plugin/Defaults.pm
+++ b/lib/Reply/Plugin/Defaults.pm
@@ -21,6 +21,7 @@ sub new {
     my $self = $class->SUPER::new(@_);
     $self->{quit} = 0;
     $self->{env} = {};
+    $self->{package} = 'main';
 
     return $self;
 }
@@ -53,8 +54,7 @@ sub compile {
         %$default_env,
     };
 
-    my $package = delete $args{package} || 'main';
-    my $prefix = "package $package;\n$PREFIX";
+    my $prefix = "package $self->{package};\n$PREFIX";
 
     return eval_closure(
         source      => "sub {\n$prefix;\n$line\n}",
@@ -70,6 +70,12 @@ sub lexical_environment {
     $self->{env}{$name} = $env;
 }
 
+sub package {
+    my $self = shift;
+    my ($package) = @_;
+    $self->{package} = $package;
+}
+
 sub execute {
     my $self = shift;
     my ($next, $code, @args) = @_;