Fixed the little peice of code.