bug fix

add to IntelliJ