Added a generic type to remove warning from eclipse