diff --git a/icecube_data_reader/events.py b/icecube_data_reader/events.py
index cb08c4f..d670806 100644
--- a/icecube_data_reader/events.py
+++ b/icecube_data_reader/events.py
@@ -8,6 +8,7 @@
 from pathlib import Path
 import numpy as np
 import numpy.typing as npt
+from typing import Iterable
 from astropy.coordinates import SkyCoord
 from astropy import units as u
 from astropy.time import Time
@@ -124,7 +125,7 @@ def __init__(
         self,
         energies: u.GeV,
         coords: SkyCoord,
-        types: np.ndarray,
+        types: Iterable[EventType],
         ang_errs: u.deg,
         mjd: Time,
     ):
@@ -194,6 +195,7 @@ def remove(self, i: int) -> None:
         self._types = np.delete(self._types, i)
         self._ang_errs = np.delete(self._ang_errs, i)
         self._mjd = np.delete(self._mjd, i)
+        self._int_types = np.delete(self._int_types, i)
 
     def to_file(
         self,