diff --git a/pkgs/development/python-modules/protobuf.nix b/pkgs/development/python-modules/protobuf.nix
index 8b9d1ddb03c..f2f0f5025d1 100644
--- a/pkgs/development/python-modules/protobuf.nix
+++ b/pkgs/development/python-modules/protobuf.nix
@@ -1,5 +1,5 @@
 { stdenv, python, buildPythonPackage
-, protobuf, google_apputils
+, protobuf, google_apputils, pyext
 , disabled, doCheck ? true }:
 
 with stdenv.lib;
@@ -9,7 +9,7 @@ buildPythonPackage rec {
   inherit disabled doCheck;
 
   propagatedBuildInputs = [ protobuf google_apputils ];
-  buildInputs = [ google_apputils ];
+  buildInputs = [ google_apputils pyext ];
 
   prePatch = ''
     while [ ! -d python ]; do
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index acb9cd141e9..dbec6e2dc79 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -19316,12 +19316,11 @@ in {
 
   protobuf = self.protobuf2_6;
   # only required by tensorflow
-  protobuf3_0_0b2 = callPackage ../development/python-modules/protobuf.nix {
-    disabled = isPyPy || isPy3k;
-    doCheck = isPy3k;
-    protobuf = pkgs.protobuf3_0_0b2;
+  protobuf3_2 = callPackage ../development/python-modules/protobuf.nix {
+    disabled = isPyPy;
+    doCheck = !isPy3k;
+    protobuf = pkgs.protobuf3_2;
   };
-
   protobuf3_0 = callPackage ../development/python-modules/protobuf.nix {
     disabled = isPyPy;
     doCheck = isPy3k;