Quick update to API import