fixes another error on same code