lib/cacheflow/memcached.rb in cacheflow-0.2.0 vs lib/cacheflow/memcached.rb in cacheflow-0.2.1

@@ -21,7 +21,11 @@
       end
     end
   end
 end
 
-Dalli::Server.prepend(Cacheflow::Memcached::Notifications)
+if defined?(Dalli::Protocol::Binary)
+  Dalli::Protocol::Binary.prepend(Cacheflow::Memcached::Notifications)
+else
+  Dalli::Server.prepend(Cacheflow::Memcached::Notifications)
+end
 Cacheflow::Memcached::Instrumenter.attach_to(:memcached)