Ahh forgetting to fix portion of the code base for issue #3.