Light refactoring

Corrected plugin reference in the main class.