Fixes the key reference