Code tweak