The fix was three lines: