Fix inference comments and warning message