Small fix to prior commit