Fix importing swappable models