Fixed test and made Test 4 pass