Fix task missing bug