Update tests to check that row count is correct post-insertion

Remove unused import