Fixed test cases to stop spitting out errors