Made change to be more like surrounding code