Fix #2
Signed-off-by: Paul Beeler <sparksco@gmail.com>