Убран баг с return