Fixed a regression