Fix randomly failing spec