Another test fix to reduce memory