Corrected accidentially removed code part in last commit