Fixes optional key types