tries to grab the correct stage when editing a task