Remove unneeded import to fix tests