Add previously unneeded import