diff --git a/Gemfile b/Gemfile
index 8ba4877..7e6a29e 100644
--- a/Gemfile
+++ b/Gemfile
@@ -2,5 +2,4 @@ source "https://rubygems.org/"
 
 gemspec
 
-# Temporary use this for module_function decorator support
-gem "yard", github: "mrkn/yard", branch: "module_function_decorator"
+gem "yard"