Fix some silly code