Adjusting code for use with another tweak