Small fix to closures