Made a simple fix