Small code improvement

Thanks @symlock
https://twitter.com/symlock/status/342893469053902848