Fix type error in assembly plugin