Use correct type for actual code