Add production check to stages