Added missing bugfix