Small fix in runtime