fix(stages): Fix loading stages in Tasks module