Just a tiny modification