Fix None test case