Small code improvement

Signed-off-by: Dmitriy Zaporozhets <dmitriy.zaporozhets@gmail.com>