Fix the use of this class in previous migrations