Minor change to complete test_optimize