Made a minor fix