Fixing bug for Kouzma