Fix failing StorageUnit test