Fix introduced bug