Fix compilation error caused by the change in IDEA API