better debugging output

[scpubgit/DKit.git] / lib / DX / SetOver.pm

diff --git a/lib/DX/SetOver.pm b/lib/DX/SetOver.pm
index 30d7516..2707dab 100644 (file)
--- a/lib/DX/SetOver.pm
+++ b/lib/DX/SetOver.pm
@@ -3,6 +3,8 @@ package DX::SetOver;
 use DX::ArrayStream;
 use Moo;
 
+with 'DX::Role::Set';
+
 has over => (is => 'ro', required => 1);
 
 has values => (is => 'ro', default => sub { {} });
@@ -40,7 +42,7 @@ sub set_value {
 
 sub remove_value {
   my ($self, $value) = @_;
-  delete $self->values->{$value->${\$self->over}};
+  delete $self->values->{ref($value) ? $value->${\$self->over} : $value};
   return $self;
 }