Is your feature request related to a problem? Please describe.
At the moment, we only have integration tests which doesn't cover all cases.

Describe the solution you'd like
We should have unit tests and eventually have 100% coverage