test: update imports