Fix failing test by wrapping in runs block