bug fix

Signed-off-by: Andrew Pinzler <pinzler@gmail.com>