Выловил баги - теперь вроде все работает