More fixes thanks to Kulcsar Dalma.