Little type comparsion fix