Fixed small flag passing error