Improve inference