Bug fix for last commit. Ugh. Sorry.