diff --git a/exercises/two-bucket/example.py b/exercises/two-bucket/example.py
index f27380e113..68435e6abd 100644
--- a/exercises/two-bucket/example.py
+++ b/exercises/two-bucket/example.py
@@ -5,10 +5,9 @@
 '''
 
 
-def two_bucket(bucket_one_cap, bucket_two_cap, desired_liters, first):
-    sizes = [bucket_one_cap, bucket_two_cap]
-    goal = desired_liters
-    goalIndex = 0 if first == 'one' else 1
+def measure(bucket_one, bucket_two, goal, start_bucket):
+    sizes = [bucket_one, bucket_two]
+    goalIndex = 0 if start_bucket == 'one' else 1
 
     def empty(buckets, i):
         return [0, buckets[1]] if i == 0 else [buckets[0], 0]
diff --git a/exercises/two-bucket/two_bucket.py b/exercises/two-bucket/two_bucket.py
index 734cf2e246..7029bcb722 100644
--- a/exercises/two-bucket/two_bucket.py
+++ b/exercises/two-bucket/two_bucket.py
@@ -1,2 +1,2 @@
-def two_bucket(bucket_one_cap, bucket_two_cap, desired_liters, first):
+def measure(bucket_one, bucket_two, goal, start_bucket):
     pass
diff --git a/exercises/two-bucket/two_bucket_test.py b/exercises/two-bucket/two_bucket_test.py
index 8a1facc6b4..1f477802e8 100644
--- a/exercises/two-bucket/two_bucket_test.py
+++ b/exercises/two-bucket/two_bucket_test.py
@@ -1,33 +1,28 @@
-'''
-All test cases adapted from
-exercism/problem-specifications/exercises/two-bucket/canonical-data.json
-'''
-
 import unittest
 
-from two_bucket import two_bucket
+from two_bucket import measure
 
 
-# Tests adapted from `problem-specifications//canonical-data.json` @ v1.2.0
+# Tests adapted from `problem-specifications//canonical-data.json` @ v1.4.0
 
 class TwoBucketTest(unittest.TestCase):
     def test_bucket_one_size_3_bucket_two_size_5_start_with_bucket_one(self):
-        self.assertEqual(two_bucket(3, 5, 1, "one"), (4, "one", 5))
+        self.assertEqual(measure(3, 5, 1, "one"), (4, "one", 5))
 
     def test_bucket_one_size_3_bucket_two_size_5_start_with_bucket_two(self):
-        self.assertEqual(two_bucket(3, 5, 1, "two"), (8, "two", 3))
+        self.assertEqual(measure(3, 5, 1, "two"), (8, "two", 3))
 
     def test_bucket_one_size_7_bucket_two_size_11_start_with_bucket_one(self):
-        self.assertEqual(two_bucket(7, 11, 2, "one"), (14, "one", 11))
+        self.assertEqual(measure(7, 11, 2, "one"), (14, "one", 11))
 
     def test_bucket_one_size_7_bucket_two_size_11_start_with_bucket_two(self):
-        self.assertEqual(two_bucket(7, 11, 2, "two"), (18, "two", 7))
+        self.assertEqual(measure(7, 11, 2, "two"), (18, "two", 7))
 
     def test_bucket_one_size_1_bucket_two_size_3_start_with_bucket_two(self):
-        self.assertEqual(two_bucket(1, 3, 3, "two"), (1, "two", 0))
+        self.assertEqual(measure(1, 3, 3, "two"), (1, "two", 0))
 
     def test_bucket_one_size_2_bucket_two_size_3_start_with_bucket_one(self):
-        self.assertEqual(two_bucket(2, 3, 3, "one"), (2, "two", 2))
+        self.assertEqual(measure(2, 3, 3, "one"), (2, "two", 2))
 
 
 if __name__ == '__main__':