fix testing bug