Fix type hint for constructor argument