Small code reduce fix