Update unit test for App