Fix some test cases not working.

Signed-off-by: Eric Taucher <Researcher0x00@gmail.com>