fixed some type complaint