trying to fix Type bug