Workaround test model loading issues