fix for the memorization task