Fix for failing spec