Fixed setup where context is not found properly.