Fix block not being passed correctly