Little fix for stages