[BUGFIX beta] Fix imports for sanitized attribute tests.