Yep, I forgot the test cases because I'm the best. Here they are fixed.