Chaning plugin initialization to use the container