Little fix for exception