Update to use the latest version of Malai