Fix default main handler with ws_init