Small fix to changes in interface