Made some minor corrections to the tests