Fixing small code-breaking error