Unit and Integration tests working fine