diff --git a/tensorflow/lite/micro/examples/magic_wand/train/requirements.txt b/tensorflow/lite/micro/examples/magic_wand/train/requirements.txt
index c83b8a48eb0..78cd89b0321 100644
--- a/tensorflow/lite/micro/examples/magic_wand/train/requirements.txt
+++ b/tensorflow/lite/micro/examples/magic_wand/train/requirements.txt
@@ -1,2 +1,2 @@
 numpy==1.16.2
-tensorflow==2.0.0-beta1
+tensorflow==2.1.4-beta1