Enable checking for stages