fix for refactored api