Bug Fixes

2 quick bug fixes I found during my use of the tool.