Fixed the "null" bug