Fixes breakage introduced by refactor
:fire:
Type