Fix #8 - Sorry for the delay, this was an easy fix but I forgot about it :(

Signed-off-by: Ross Allan <rallanpcl@gmail.com>