Fix for broken code