Fixed type error @ line 36