update to work with v0.4-dev