Fix for tests

Updated spec to prevent test failure