update the entry point class name
Signed-off-by: Haozi Leung <haozileung.cn@gmail.com>