Modify tests with context parameter