Fix/remove faulty import during migrations