Added missing class movement