Fixed context on unit test