Fix broken test

@stephensanchez