Updated to make tests run properly again