Remove unsafety in test