minor fix

Fix to apply ozzy's latest comment