Correct migration TypeError