Edited a test to get it to work.