bug fixes

Signed-off-by: cout970 <thecout970@gmail.com>