Fixed two of the tests in enkf_test.py