Fixes shifting classes, I'm sure this isn't the best way to do it but seems logical for now.