Remove un-necessary call to get input dataset