Repair unit test

Signed-off-by: Adam Engebretson <adam@enge.me>