fixed error from testing