fix checks to use proper Enum and not String