I believe I actually know the answer, it lies in your classpath of the module in your gradle settings when you go to run/debug your configurations
Mine works fine if I set it to my main class module