Fix error in context part