lib/verdict/storage/memory_storage.rb in verdict-0.2.1 vs lib/verdict/storage/memory_storage.rb in verdict-0.3.0

@@ -18,11 +18,11 @@
         experiment_store = @assignments[experiment.handle] || {}
         experiment_store[subject_identifier]
       end
 
       def remove_assignment(experiment, subject_identifier)
-        @assignments[assignment.experiment.handle] ||= {}
-        @assignments[assignment.experiment.handle].delete(subject_identifier)
+        @assignments[experiment.handle] ||= {}
+        @assignments[experiment.handle].delete(subject_identifier)
       end
 
       def clear_experiment(experiment)
         @assignments.delete(experiment.handle)
       end