[FIX] Tests not working with travis due to 'ImportError: cannot import name api'. Try to fix the tests by avoiding the direct import