Add assembly plugin back