diff --git a/Dockerfile b/Dockerfile
index 669555c..d010733 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,7 +4,7 @@ LABEL "email"="dionyself@gmail.com"
 RUN apk add --no-cache git gcc g++
 RUN mkdir /app
 RUN cd /app && git clone https://github.com/dionyself/golang-cms && cd golang-cms \
- && go get -u github.com/beego/bee \
+ && go get github.com/beego/bee/v2 \
  && go mod tidy
 
 EXPOSE 8080
diff --git a/run.sh b/run.sh
index fb44036..4cd5551 100644
--- a/run.sh
+++ b/run.sh
@@ -4,5 +4,6 @@ cd /app/golang-cms
 git pull origin master --tags
 export GOLANG_CMS_VERSION=$(git tag --sort=committerdate | tail -1)
 git checkout $GOLANG_CMS_VERSION
+go get -u github.com/beego/bee/v2
 go mod tidy
 bee run