fix bug

Signed-off-by: Alexandr <xescoder@gmail.com>