bug fixing
Signed-off-by: GameOwen <gameowencz@gmail.com>