Fix IDEA issue