Add tests for non-happy path