Update to work with version 1.7.2