Fixes #2

Signed-off-by: Euan Torano <euantorano@gmail.com>