[misc] Fixed broken code