Fixes

Signed-off-by: MishkaRogachev <mishkarogachev@gmail.com>