fix #4 IDEA probrem