Fixed the head bug