Moved test to spec