Modified the tests one failing