Fix test, with with statement