Fixed node correction so it should work fine

Solution may be changed in future