Fixed error introduce in last commit