update main test

Signed-off-by: Nicholas Lum <nicholaslum444@hotmail.com>