Just a short modification