Made corrections to tests