Fixed minor type bug