Фикс баги

Пофиксил небольшую багу, теперь всё должно компилиться