bug fixing

Signed-off-by: tdaffron <tdaffron@gmail.com>