Minor bugfixed