Fix failing specs (due to bad nil reference)