Fix for fix :D

Minor fix