diff --git a/tb_plugin/torch_tb_profiler/__init__.py b/tb_plugin/torch_tb_profiler/__init__.py
index d70660903..c94e71bf0 100644
--- a/tb_plugin/torch_tb_profiler/__init__.py
+++ b/tb_plugin/torch_tb_profiler/__init__.py
@@ -4,4 +4,4 @@
 
 # Entry point for Pytorch TensorBoard plugin package.
 
-__version__ = '0.3.1'
+__version__ = '0.4.0'