Adding small fix by Alex Puterbaugh. Thanks.