Fix tests by removing PunktWordTokenizer import