Whoops missed this function for earlier commit