Fix broken test caused by modifying how the generateLock task works