Add some prediction to remove function call