Resolves #4

Easy correct fix for this problem, I should've done this previously