Fixed a newly introduced bug