Fixed minor bug in test cases

I put up old versions originally