Minor code fix with naming convention