updated v0.96.0 withmodule implementation