fixed unit test so it didnt produce an error