Fixed test (this stuff is really shaky)