updated to work with version 0.7.2