Small fix in the framework