Fix docker issue