diff --git a/bin/container.sh b/bin/container.sh
index 02620ef84..0c3ee9fc6 100755
--- a/bin/container.sh
+++ b/bin/container.sh
@@ -184,7 +184,7 @@ start_cassandra() {
     IS_IMAGE_AVAILABLE=$?
     if [ "${migration}" = true ] || ! [ ${IS_IMAGE_AVAILABLE} -eq 0 ]; then
         echo "${green}${project}: [Start] Building wasabi keyspace image${reset}"
-        docker build -t wasabi-keyspace:latest -f "${CURRENT_DIR}/./docker/cqlsh.docker" "${CURRENT_DIR}/./docker/"
+        docker build --force-rm --no-cache -t wasabi-keyspace:latest -f "${CURRENT_DIR}/./docker/cqlsh.docker" "${CURRENT_DIR}/./docker/"
     fi
     docker run -it --rm -e CASSANDRA_KEYSPACE_PREFIX=${project} -e CQLSH_HOST=${project}-cassandra -e CASSANDRA_PORT=9042 --net=${docker_network} --name wasabi_create_keyspace wasabi-keyspace
 
@@ -192,7 +192,7 @@ start_cassandra() {
     IS_IMAGE_AVAILABLE=$?
     if [ "${migration}" = true ] || ! [ ${IS_IMAGE_AVAILABLE} -eq 0 ]; then
         echo "${green}${project}: [Start] Building wasabi migration image${reset}"
-        docker build -t wasabi-migration:latest -f "${CURRENT_DIR}/./docker/migration.docker" "${CURRENT_DIR}/../"
+        docker build --force-rm --no-cache -t wasabi-migration:latest -f "${CURRENT_DIR}/./docker/migration.docker" "${CURRENT_DIR}/../"
     fi
     docker run -it --rm -e CQLSH_HOST=${project}-cassandra -e CASSANDRA_PORT=9042 --net=${docker_network} --name wasabi_migration wasabi-migration
     echo "${green}${project}: [DONE] creating keyspace and migration schemas${reset}"
diff --git a/bin/docker/migration.docker b/bin/docker/migration.docker
index b17d755b9..75bb2d112 100644
--- a/bin/docker/migration.docker
+++ b/bin/docker/migration.docker
@@ -2,7 +2,7 @@ FROM openjdk:jre-alpine
 
 
 ADD https://oss.sonatype.org/content/repositories/public/com/builtamont/cassandra-migration/0.9/cassandra-migration-0.9-jar-with-dependencies.jar /wasabi/cassandra-migration.jar
-ADD modules/repository-datastax/db/mutation /wasabi/mutation
+ADD modules/repository-datastax/src/main/resources/com/intuit/wasabi/repository/impl/cassandra/migration /wasabi/mutation
 ADD bin/docker/schema_migration.sh /wasabi/migration.sh
 
 ENTRYPOINT ["/wasabi/migration.sh"]