Return given value value in Module.put_attribute/3 (#5218)

2 files changed, 7 insertions, 0 deletions

diff --git a/lib/elixir/lib/module.ex b/lib/elixir/lib/module.ex
index 06905a028..36c840cad 100644
--- a/lib/elixir/lib/module.ex
+++ b/lib/elixir/lib/module.ex
@@ -904,6 +904,7 @@ defmodule Module do
       end
 
     :ets.insert(table, {key, new})
+    value
   end
 
   @doc """
diff --git a/lib/elixir/test/elixir/module_test.exs b/lib/elixir/test/elixir/module_test.exs
index 43e4834ee..483e3644f 100644
--- a/lib/elixir/test/elixir/module_test.exs
+++ b/lib/elixir/test/elixir/module_test.exs
@@ -58,6 +58,12 @@ defmodule ModuleTest do
     end
   end
 
+  test "return value" do
+    in_module do
+      assert (@return [:foo, :bar]) == [:foo, :bar]
+    end
+  end
+
   test "in memory" do
     assert :code.which(__MODULE__) == :in_memory
   end