Fix tests to work in IntelliJ