Fixed the geometry shader