Fixed a small regression