the key feature need more time than expected.
The normal way work(visual), but after game restart your keys decrease and if i changed the modded diamond to keys the game force closed.
At the moment i wait of a new update from the dev and try then again.