lib/option.rb in option-0.4.0 vs lib/option.rb in option-0.5.0

@@ -55,10 +55,10 @@
   def empty?
     false
   end
 
   def map(&blk)
-    Option(blk.call(get))
+    self.class.new(blk.call(get))
   end
 
   def flat_map(&blk)
     assert_option(blk.call(get))
   end