Adding correct type