Fixed a broken code