I think adding a closure would be better (like you did in the other test files)