Implement two stages to distinguish between stage and prod