Fix ruby block for only_if