Modded some test cases