Updates test to pass