Fix unit tests

@RaymondLim 

Fix the native menu unit tests.