Fix failing test

@braddunbar can you confirm this is expected behavior?