Fix comment and __construct param