Bug fix

Signed-off-by: Tom Castleman <tom.castleman@b3-it.com>