Fixed a small type error