Adds null checking for keys

Stupid customItemStack