diff --git a/stages/01-Baseimage/01-run.sh b/stages/01-Baseimage/01-run.sh
index 9c0b487..23c3482 100644
--- a/stages/01-Baseimage/01-run.sh
+++ b/stages/01-Baseimage/01-run.sh
@@ -28,7 +28,7 @@ pushd ${STAGE_WORK_DIR}
     echo "bytes"
     ls -l
 
-if [[ "${OS}" != ubuntu-x86 ]] && [[ "${OS}" != debian-X20 ]]; then
+if [[ "${OS}" != ubuntu-x86 ]] && [[ "${OS}" != debian-X20 ]] && [[ "${OS}" != ubuntu ]]; then
 
     log "Create empty image" #this will be attached to the base image to increase the size of it
     dd if=/dev/zero of=temp.img bs=1 count=1 seek=$DIFFERENCE