bug fix: MWS not accept stage as function