Fixed messed up code.