Fixes a mistake in not allowing custom keys.