fixed import again

didn't quite fix it last time, but I got it this time