Fix the unmount args in the test