fix type
minor changes