Fix feature tests

- Fix feature tests for investigation count.