update for the latest version of malai