fix finnal bug