fixed bug to previous commit