Fix add key controls