Closes #31. Pretty simple fix.