Fixed little oversights