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