Code sniff fix