Fixed a slight bug introduced in the last commit