Fix imports for standalone memory benchmark scripts