Fix test

Signed-off-by: Jonathan Rudenberg <jonathan@titanous.com>