Fixed some red code