Small testing fix