Fix issue #45

Signed-off-by: Max Fierke <max@maxfierke.com>