Forgot to remove unused package import