Fix some mistakes I made