fix bug

Signed-off-by: fantaros <fantasymon@gmail.com>