Fix test for new memory limit