Changing the code according new impl.

Ref: https://github.com/specta/specta/issues/105