Minor impt tweak