Baseline unit test now working