Fix code leak