Fix shell integration tests

I failed to validate this as I'm getting the following error (not sure it's related), but if anyone wants to pick up from here:
```

Checking row count in database table matches import from file...
Expected 292 does not match actual value (0)


```