fix breakage from import cleanup