We should have more robust Unit Tests that we can run in the CI pipeline (like the main Toolkit repo has now).