Fixed the test cases

Paiseh I fixed something after doing the test cases then I forgot to
update it.