Made a small tweak