I have an issue where, if I create this as main.py, I will get an error when I try to start my M5 paper. The error is Wifi Invalid Mode.

My script is below. I have the same configuration in M5 burner and it seems to correctly trying to start main.py as I get the error straight away. If I chose to show the menu at start up then I can see the M5 paper in UIFlow 2 and I can see the device has connected to my account. Any ideas? It's a bit of a pain having choose the app from the applist on the devices screen, but I can run it like that. My app is a bit more complicated than this but this script does the same thing:

import os, sys, io
import M5
from M5 import *
import network

label0 = None
wlan = None

def setup():
global label0, wlan

wlan = network.WLAN(network.STA_IF)
M5.begin()
Widgets.fillScreen(0xeeeeee)
label0 = Widgets.Label("label0", 124, 542, 1.0, 0xffffff, 0x000000, Widgets.FONTS.DejaVu18)

wlan.connect('network', 'testtesttest')
label0.setFont(Widgets.FONTS.DejaVu24)
label0.setText(str(wlan.ifconfig()[0]))

def loop():
global label0, wlan
M5.update()

if name == 'main':
try:
setup()
while True:
loop()
except (Exception, KeyboardInterrupt) as e:
try:
from utility import print_error_msg
print_error_msg(e)
except ImportError:
print("please update to latest firmware")

Thanks,