Tests update

One assertion wasn't wrapped in an It statement; fixed.