Kate Winslet returns to filming after falling on the set of Lee