Adding a special case for staging machines