Minor memory buffer fix