Use correct key type