Updates to Unity test