Fix imports for standalone memory benchmark scripts

Minor import issue I ran into when playing with the provided benchmark scripts.