minor changes in inference interface