Fixed compilatin issue from last commit