Change WBStage interface for consistency.