Minor fix becuase of API change in spec