Fix unhashable type issue