Small style improvement