diff --git a/.github/workflows/doc-build.yml b/.github/workflows/doc-build.yml
index 21a74991..80332bf8 100644
--- a/.github/workflows/doc-build.yml
+++ b/.github/workflows/doc-build.yml
@@ -35,7 +35,7 @@ jobs:
           cd optimum
           version=`echo "$(grep '^__version__ =' tpu/version.py | cut -d '=' -f 2- | xargs)"`
 
-          if [[ $version == *a* ]]
+          if [[ $version == *.dev0 ]]
           then
             echo "VERSION=main" >> $GITHUB_ENV
           else
diff --git a/optimum/tpu/version.py b/optimum/tpu/version.py
index dc6e5f1c..315159c2 100644
--- a/optimum/tpu/version.py
+++ b/optimum/tpu/version.py
@@ -15,5 +15,5 @@
 from packaging.version import parse
 
 
-__version__ = "0.2.3"
+__version__ = "0.2.3.dev0"
 VERSION = parse(__version__)
diff --git a/text-generation-inference/docker/Dockerfile b/text-generation-inference/docker/Dockerfile
index 7d3faf21..d299a307 100644
--- a/text-generation-inference/docker/Dockerfile
+++ b/text-generation-inference/docker/Dockerfile
@@ -81,7 +81,7 @@ RUN apt-get update -y \
 RUN pip3 --no-cache-dir install --upgrade pip
 
 ARG ENABLE_GOOGLE_FEATURE
-ARG VERSION='0.2.3'
+ARG VERSION='0.2.3.dev0'
 RUN test -n ${VERSION:?}
 
 FROM base AS optimum-tpu-installer
diff --git a/text-generation-inference/integration-tests/conftest.py b/text-generation-inference/integration-tests/conftest.py
index 3365cfd1..7b7ef2ca 100644
--- a/text-generation-inference/integration-tests/conftest.py
+++ b/text-generation-inference/integration-tests/conftest.py
@@ -35,7 +35,7 @@ def validate_ci_tpu_env_format(env_string: str) -> bool:
     """
     Validate that CI TPU environment string follows '--env Argument' pattern.
     Returns True if valid, False otherwise.
-    """        
+    """
     parts = env_string.split()
     return len(parts) % 2 == 0 and all(
         parts[i] == "--env" and not parts[i + 1].startswith("--env")
@@ -49,7 +49,7 @@ def process_ci_tpu_env_vars(env_string: str) -> dict:
     env_vars = {}
     # Extract variables from string
     tpu_vars = [x.strip() for x in env_string.split('--env') if x.strip()]
-    
+
     # Process each variable
     for var in tpu_vars:
         env_value = os.environ.get(var, "")
@@ -57,7 +57,7 @@ def process_ci_tpu_env_vars(env_string: str) -> dict:
         # Log if environment variable is not set
         if not env_value:
             logger.warning(f"TPU environment variable {var} is not set")
-            
+
     return env_vars
 
 
@@ -212,12 +212,12 @@ def docker_launcher(
 
         # Add TPU environment variables when running in CI
         if TPU_ENV:
-            logger.info(f"TPU_ENV is set, adding specific TPU environment variables for the CI")
+            logger.info("TPU_ENV is set, adding specific TPU environment variables for the CI")
             logger.debug(f"TPU_ENV: {TPU_ENV}")
             # Validate TPU environment format
             if not validate_ci_tpu_env_format(TPU_ENV):
                 raise ValueError("Invalid TPU environment format", TPU_ENV)
-                
+
             # Process TPU environment variables
             tpu_env_vars = process_ci_tpu_env_vars(TPU_ENV)
             env.update(tpu_env_vars)
diff --git a/text-generation-inference/server/text_generation_server/version.py b/text-generation-inference/server/text_generation_server/version.py
index d81884df..78f2318b 100644
--- a/text-generation-inference/server/text_generation_server/version.py
+++ b/text-generation-inference/server/text_generation_server/version.py
@@ -1,5 +1,5 @@
 from pkg_resources import parse_version
 
 
-__version__ = "0.2.3"
+__version__ = "0.2.3.dev0"
 VERSION = parse_version(__version__)