Fix EntityAPI dependency