tiny fix, more concise and should be faster