Fixed bug w/ incorrect type