Fix failing test, thanks @ericmj