Set correct type