This looks like a simple fix