minor problem in inference module