Fixed unit test snafu