Fixed inference issue #2