Hey @yashodharanawaka,
The interface you are implementing is not part of the public API. In order for us to be able to help you more we would need a Minimal Reproducible example. Can you please create a small project and share it with us. This way we can run exactly what you are doing and we can see what the root cause of the problem is.
Cheers,
Filip