IntelliJ Changes to make unit tests work