Fix for bad code change