Lab4. Minor bugfix