Fix for wrong variable