Fix failing test.

Signed-off-by: Nick Campbell <nicholas.j.campbell@gmail.com>