fixed the coding fail from last commit