We need to change the return value here if it returns a promise