fix (need testing)

Signed-off-by: Konctantin <gawrilyako@gmail.com>