Fixes #3

Signed-off-by: Jacob Torrey <discipleofranok@gmail.com>