Fix bad code due to half ready IntelliJ, sorry