Trying to fix unit test

DO NOT MERGE